package com.yoshino.flyweight;

import java.util.HashMap;

/**
 * 网站工厂类，根据需求返回一个网站
 * @author yoshino
 **/
public class WebSiteFactory {

    // 集合，充当池

    private HashMap<String, WebSite> pool = new HashMap<>();

    // 根据网站的类型，返回一个网站，如果没有就创建一个网站，并放入到池中

    public WebSite getWebSiteCategory(String type) {
        if (!pool.containsKey(type)) {
            // 就创建一个网站，并放入池中
            pool.put(type, new ConcreteWebSite(type));
        }
        return pool.get(type);
    }

    public int getWebSiteCount() {
        return pool.size();
    }
}
